| Name | Version | Summary | date |
| image-classification-jax |
0.1.4 |
Run image classification experiments in JAX with ViT, resnet, cifar10, cifar100, imagenette, and imagenet. |
2024-12-18 23:59:49 |
| reinforced-lib |
1.1.5 |
Reinforcement learning library |
2024-12-18 12:09:00 |
| invrs-opt |
0.10.1 |
Algorithms for inverse design |
2024-12-18 00:39:45 |
| metrx |
0.1.0 |
"A library containing a collection of distance and similarity measures to compare time series data." |
2024-12-16 12:59:14 |
| lipsutils |
0.0.3 |
Internal utilities shared by the Princeton LIPS group. |
2024-12-14 18:45:54 |
| differt-core |
0.0.30 |
Core backend of DifffeRT implemented in Rust |
2024-12-13 17:28:19 |
| DiffeRT |
0.0.30 |
Differentiable Ray Tracing Toolbox for Radio Propagation Simulations |
2024-12-13 17:25:21 |
| spbessax |
0.1.3 |
spherical Bessel functions for JAX |
2024-12-13 11:08:21 |
| invrs-gym |
1.4.4 |
A collection of inverse design challenges |
2024-12-12 17:05:25 |
| fibermc |
0.0.4 |
A Jax-based differentiable Monte Carlo estimator with applications to differentiable simulation, computational geometry, and topology optimization. |
2024-12-10 19:42:08 |
| cbfpy |
0.0.1 |
Control Barrier Functions in Python |
2024-12-10 06:25:42 |
| diffrax |
0.6.1 |
GPU+autodiff-capable ODE/SDE/CDE solvers written in JAX. |
2024-12-10 03:39:37 |
| psgd-jax |
0.2.8 |
An implementation of PSGD optimizer in JAX. |
2024-12-08 20:46:32 |
| equinox |
0.11.10 |
Elegant easy-to-use neural networks in JAX. |
2024-12-08 02:44:40 |
| qarray |
1.4.4 |
Qarray, a GPU accelerated quantum dot array simulator, leveraging parallelised Rust and JAX XLA acceleration |
2024-12-07 12:22:45 |
| fmmax |
0.11.0 |
Fourier modal method with Jax |
2024-12-07 00:00:47 |
| triton-model-navigator |
0.13.0 |
Triton Model Navigator: An inference toolkit for optimizing and deploying machine learning models and pipelines on the Triton Inference Server and PyTriton. |
2024-12-06 14:40:12 |
| cryojax |
0.3.2 |
Cryo-EM image simulation and analysis powered by JAX |
2024-12-06 04:05:34 |
| tridiax |
0.2.1 |
Solving tridiagonal systems. |
2024-12-05 09:55:54 |
| ott-jax |
0.5.0 |
Optimal Transport Tools in JAX |
2024-12-03 14:37:54 |